from ai import AiConstant
from ai.utils import utils_file

DATA_DIR = AiConstant.DATA_PATH + 'gpt/data/novel/'
OUTPUT_DIR = AiConstant.OUTPUT_PATH + 'system_learning/huggingface/novel_pretrain_generator/'
VOCAB_DIR = AiConstant.DATA_PATH + 'gpt/vocab/'
LOG_DIR = AiConstant.LOG_PATH + 'system_learning/huggingface/novel_pretrain_generator/'

for file in [DATA_DIR, OUTPUT_DIR, VOCAB_DIR, LOG_DIR]:
    utils_file.makedir_for_file_or_dir(file)

logger = AiConstant.AI_LOGGER(LOG_DIR + 'novel.log')


def get_logger():
    return logger


#  超参数
BATCH_SIZE = 12
LEARN_RATE = 1e-3
EPOCHS = 1
